home *** CD-ROM | disk | FTP | other *** search
Wrap
# Source Generated with Decompyle++ # File: in.pyc (Python 2.4) import rfc822 import sys import unittest from test import test_support try: from cStringIO import StringIO except ImportError: from StringIO import StringIO class MessageTestCase(unittest.TestCase): def create_message(self, msg): return rfc822.Message(StringIO(msg)) def test_get(self): msg = self.create_message('To: "last, first" <userid@foo.net>\n\ntest\n') self.assert_(msg.get('to') == '"last, first" <userid@foo.net>') self.assert_(msg.get('TO') == '"last, first" <userid@foo.net>') self.assert_(msg.get('No-Such-Header') is None) self.assert_(msg.get('No-Such-Header', 'No-Such-Value') == 'No-Such-Value') def test_setdefault(self): msg = self.create_message('To: "last, first" <userid@foo.net>\n\ntest\n') self.assert_(not msg.has_key('New-Header')) self.assert_(msg.setdefault('New-Header', 'New-Value') == 'New-Value') self.assert_(msg.setdefault('New-Header', 'Different-Value') == 'New-Value') self.assert_(msg['new-header'] == 'New-Value') self.assert_(msg.setdefault('Another-Header') == '') self.assert_(msg['another-header'] == '') def check(self, msg, results): '''Check addresses and the date.''' m = self.create_message(msg) i = 0 for n, a in m.getaddrlist('to') + m.getaddrlist('cc'): try: mn = results[i][0] ma = results[i][1] except IndexError: print 'extra parsed address:', repr(n), repr(a) continue i = i + 1 if mn == n and ma == a: continue print 'not found:', repr(n), repr(a) out = m.getdate('date') if out: self.assertEqual(out, (1999, 1, 13, 23, 57, 35, 0, 1, 0), 'date conversion failed') def test_basic(self): self.check('Date: Wed, 13 Jan 1999 23:57:35 -0500\nFrom: Guido van Rossum <guido@CNRI.Reston.VA.US>\nTo: "Guido van\n\t : Rossum" <guido@python.org>\nSubject: test2\n\ntest2\n', [ ('Guido van\n\t : Rossum', 'guido@python.org')]) self.check('From: Barry <bwarsaw@python.org\nTo: guido@python.org (Guido: the Barbarian)\nSubject: nonsense\nDate: Wednesday, January 13 1999 23:57:35 -0500\n\ntest', [ ('Guido: the Barbarian', 'guido@python.org')]) self.check('From: Barry <bwarsaw@python.org\nTo: guido@python.org (Guido: the Barbarian)\nCc: "Guido: the Madman" <guido@python.org>\nDate: 13-Jan-1999 23:57:35 EST\n\ntest', [ ('Guido: the Barbarian', 'guido@python.org'), ('Guido: the Madman', 'guido@python.org')]) self.check('To: "The monster with\n the very long name: Guido" <guido@python.org>\nDate: Wed, 13 Jan 1999 23:57:35 -0500\n\ntest', [ ('The monster with\n the very long name: Guido', 'guido@python.org')]) self.check('To: "Amit J. Patel" <amitp@Theory.Stanford.EDU>\nCC: Mike Fletcher <mfletch@vrtelecom.com>,\n "\'string-sig@python.org\'" <string-sig@python.org>\nCc: fooz@bat.com, bart@toof.com\nCc: goit@lip.com\nDate: Wed, 13 Jan 1999 23:57:35 -0500\n\ntest', [ ('Amit J. Patel', 'amitp@Theory.Stanford.EDU'), ('Mike Fletcher', 'mfletch@vrtelecom.com'), ("'string-sig@python.org'", 'string-sig@python.org'), ('', 'fooz@bat.com'), ('', 'bart@toof.com'), ('', 'goit@lip.com')]) self.check('To: Some One <someone@dom.ain>\nFrom: Anudder Persin <subuddy.else@dom.ain>\nDate:\n\ntest', [ ('Some One', 'someone@dom.ain')]) self.check('To: person@dom.ain (User J. Person)\n\n', [ ('User J. Person', 'person@dom.ain')]) def test_twisted(self): self.check('To: <[smtp:dd47@mail.xxx.edu]_at_hmhq@hdq-mdm1-imgout.companay.com>\nDate: Wed, 13 Jan 1999 23:57:35 -0500\n\ntest', [ ('', ''), ('', 'dd47@mail.xxx.edu'), ('', '_at_hmhq@hdq-mdm1-imgout.companay.com')]) def test_commas_in_full_name(self): self.check('To: "last, first" <userid@foo.net>\n\ntest', [ ('last, first', 'userid@foo.net')]) def test_quoted_name(self): self.check('To: (Comment stuff) "Quoted name"@somewhere.com\n\ntest', [ ('Comment stuff', '"Quoted name"@somewhere.com')]) def test_bogus_to_header(self): self.check('To: :\nCc: goit@lip.com\nDate: Wed, 13 Jan 1999 23:57:35 -0500\n\ntest', [ ('', 'goit@lip.com')]) def test_addr_ipquad(self): self.check('To: guido@[132.151.1.21]\n\nfoo', [ ('', 'guido@[132.151.1.21]')]) def test_iter(self): m = rfc822.Message(StringIO('Date: Wed, 13 Jan 1999 23:57:35 -0500\nFrom: Guido van Rossum <guido@CNRI.Reston.VA.US>\nTo: "Guido van\n\t : Rossum" <guido@python.org>\nSubject: test2\n\ntest2\n')) self.assertEqual(sorted(m), [ 'date', 'from', 'subject', 'to']) def test_rfc2822_phrases(self): self.check('To: User J. Person <person@dom.ain>\n\n', [ ('User J. Person', 'person@dom.ain')]) def test_2getaddrlist(self): eq = self.assertEqual msg = self.create_message('To: aperson@dom.ain\nCc: bperson@dom.ain\nCc: cperson@dom.ain\nCc: dperson@dom.ain\n\nA test message.\n') ccs = [ ('', a) for a in [ 'bperson@dom.ain', 'cperson@dom.ain', 'dperson@dom.ain'] ] addrs = msg.getaddrlist('cc') addrs.sort() eq(addrs, ccs) addrs = msg.getaddrlist('cc') addrs.sort() eq(addrs, ccs) def test_parseaddr(self): eq = self.assertEqual eq(rfc822.parseaddr('<>'), ('', '')) eq(rfc822.parseaddr('aperson@dom.ain'), ('', 'aperson@dom.ain')) eq(rfc822.parseaddr('bperson@dom.ain (Bea A. Person)'), ('Bea A. Person', 'bperson@dom.ain')) eq(rfc822.parseaddr('Cynthia Person <cperson@dom.ain>'), ('Cynthia Person', 'cperson@dom.ain')) def test_quote_unquote(self): eq = self.assertEqual eq(rfc822.quote('foo\\wacky"name'), 'foo\\\\wacky\\"name') eq(rfc822.unquote('"foo\\\\wacky\\"name"'), 'foo\\wacky"name') def test_main(): test_support.run_unittest(MessageTestCase) if __name__ == '__main__': test_main()